1490G - Old Floppy Drive - CodeForces Solution


binary search data structures math *1900

Please click on ads to support us..

Python Code:

import sys

input = sys.stdin.readline
inf = float('inf')


def getInt():
    return int(input())


def getStr():
    return input().strip()


def getList(split=True):
    s = getStr()
    if split:
        s = s.split()
    return list(map(int, s))


t = getInt()


def solve():
    n, m = getList()
    a = getList()
    q = getList()
    from itertools import accumulate
    p = list(accumulate(a))
    gain = list(accumulate(p, max))
    from bisect import bisect_left, bisect
    for i in q:
        if i <= gain[-1]:
            it = bisect_left(gain, i)
            print(it, end=" ")
        elif p[-1] <= 0:
            print(-1, end=" ")
        else:
                        k = (i-gain[-1] + p[-1]-1) // (p[-1])
            i -= p[-1]*k
            res = n * k
            if i:
                res += bisect_left(gain, i)+1
            print(res-1, end=" ")
    print()


for _ in range(t):
    solve()


Comments

Submit
0 Comments
More Questions

1566B - MIN-MEX Cut
678C - Joty and Chocolate
1352E - Special Elements
1520E - Arranging The Sheep
1157E - Minimum Array
1661D - Progressions Covering
262A - Roma and Lucky Numbers
1634B - Fortune Telling
1358A - Park Lighting
253C - Text Editor
365B - The Fibonacci Segment
75A - Life Without Zeros
1519A - Red and Blue Beans
466A - Cheap Travel
659E - New Reform
1385B - Restore the Permutation by Merger
706A - Beru-taxi
686A - Free Ice Cream
1358D - The Best Vacation
1620B - Triangles on a Rectangle
999C - Alphabetic Removals
1634C - OKEA
1368C - Even Picture
1505F - Math
1473A - Replacing Elements
959A - Mahmoud and Ehab and the even-odd game
78B - Easter Eggs
1455B - Jumps
1225C - p-binary
1525D - Armchairs